import pygame
from pygame.sprite import Sprite

class Bullet(Sprite):
    """定义的飞船子弹的实体类"""

    def __init__(self, ai_setting, screen, ship):
        """该函数在飞船所在的位置创建一个子弹的对象"""
        super(Bullet, self).__init__()
        self.screen = screen

        # 使用矩形创建一个子弹的初始位置然后再根据飞船的位置画出子弹的具体位置
        self.rect = pygame.Rect(0, 0, ai_setting.bullet_width, ai_setting.bullet_height)
        self.rect.centerx = ship.rect.centerx
        self.rect.top = ship.rect.top

        # 使用小数存储子弹的初始位置
        self.y = float(self.rect.y)

        # 设置子弹的背景颜色和移动速度
        self.color = ai_setting.bullet_color
        self.speed_factor = ai_setting.bullet_speed_factor

    def update(self):
        """更新子弹的位置，子弹的位置是向上移动的"""
        # 计算当前子弹应该所在的位置（y轴上的位置）
        self.y -= self.speed_factor
        # 重新设置子弹的外接矩形的位置
        self.rect.y = self.y

    def draw_bullet(self):
        """在屏幕上画出子弹的外接矩形的位置"""
        pygame.draw.rect(self.screen, self.color, self.rect)

